// Copyright (c) 2013 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef UI_MESSAGE_CENTER_MESSAGE_CENTER_IMPL_H_
#define UI_MESSAGE_CENTER_MESSAGE_CENTER_IMPL_H_

#include <stddef.h>

#include <string>
#include <vector>

#include "base/macros.h"
#include "base/memory/scoped_ptr.h"
#include "base/memory/weak_ptr.h"
#include "base/time/time.h"
#include "base/timer/timer.h"
#include "ui/message_center/message_center.h"
#include "ui/message_center/message_center_observer.h"
#include "ui/message_center/message_center_types.h"
#include "ui/message_center/notification_blocker.h"
#include "ui/message_center/notifier_settings.h"
#include "ui/message_center/popup_timers_controller.h"

namespace message_center {
class NotificationDelegate;
class MessageCenterImpl;

namespace internal {
    class ChangeQueue;
}

// The default implementation of MessageCenter.
class MessageCenterImpl : public MessageCenter,
                          public NotificationBlocker::Observer,
                          public message_center::NotifierSettingsObserver {
public:
    MessageCenterImpl();
    ~MessageCenterImpl() override;

    // MessageCenter overrides:
    void AddObserver(MessageCenterObserver* observer) override;
    void RemoveObserver(MessageCenterObserver* observer) override;
    void AddNotificationBlocker(NotificationBlocker* blocker) override;
    void RemoveNotificationBlocker(NotificationBlocker* blocker) override;
    void SetVisibility(Visibility visible) override;
    bool IsMessageCenterVisible() const override;
    size_t NotificationCount() const override;
    size_t UnreadNotificationCount() const override;
    bool HasPopupNotifications() const override;
    bool IsQuietMode() const override;
    bool HasClickedListener(const std::string& id) override;
    message_center::Notification* FindVisibleNotificationById(
        const std::string& id) override;
    const NotificationList::Notifications& GetVisibleNotifications() override;
    NotificationList::PopupNotifications GetPopupNotifications() override;
    void AddNotification(scoped_ptr<Notification> notification) override;
    void UpdateNotification(const std::string& old_id,
        scoped_ptr<Notification> new_notification) override;
    void RemoveNotification(const std::string& id, bool by_user) override;
    void RemoveAllNotifications(bool by_user) override;
    void RemoveAllVisibleNotifications(bool by_user) override;
    void SetNotificationIcon(const std::string& notification_id,
        const gfx::Image& image) override;
    void SetNotificationImage(const std::string& notification_id,
        const gfx::Image& image) override;
    void SetNotificationButtonIcon(const std::string& notification_id,
        int button_index,
        const gfx::Image& image) override;
    void DisableNotificationsByNotifier(const NotifierId& notifier_id) override;
    void ClickOnNotification(const std::string& id) override;
    void ClickOnNotificationButton(const std::string& id,
        int button_index) override;
    void ClickOnSettingsButton(const std::string& id) override;
    void MarkSinglePopupAsShown(const std::string& id,
        bool mark_notification_as_read) override;
    void DisplayedNotification(const std::string& id,
        const DisplaySource source) override;
    void SetNotifierSettingsProvider(NotifierSettingsProvider* provider) override;
    NotifierSettingsProvider* GetNotifierSettingsProvider() override;
    void SetQuietMode(bool in_quiet_mode) override;
    void EnterQuietModeWithExpire(const base::TimeDelta& expires_in) override;
    void RestartPopupTimers() override;
    void PausePopupTimers() override;
    void ForceNotificationFlush(const std::string& id) override;

    // NotificationBlocker::Observer overrides:
    void OnBlockingStateChanged(NotificationBlocker* blocker) override;

    // message_center::NotifierSettingsObserver overrides:
    void UpdateIconImage(const NotifierId& notifier_id,
        const gfx::Image& icon) override;
    void NotifierGroupChanged() override;
    void NotifierEnabledChanged(const NotifierId& notifier_id,
        bool enabled) override;

    // Unexposed methods:
    void AddNotificationImmediately(scoped_ptr<Notification> notification);
    void UpdateNotificationImmediately(
        const std::string& old_id,
        scoped_ptr<Notification> new_notification);
    void RemoveNotificationImmediately(const std::string& id, bool by_user);

protected:
    void DisableTimersForTest() override;
    void EnableChangeQueueForTest(bool enable) override;

private:
    struct NotificationCache {
        NotificationCache();
        ~NotificationCache();
        void Rebuild(const NotificationList::Notifications& notifications);
        void RecountUnread();

        NotificationList::Notifications visible_notifications;
        size_t unread_count;
    };

    void RemoveNotifications(bool by_user, const NotificationBlockers& blockers);
    void RemoveNotificationsForNotifierId(const NotifierId& notifier_id);

    scoped_ptr<NotificationList> notification_list_;
    NotificationCache notification_cache_;
    base::ObserverList<MessageCenterObserver> observer_list_;
    scoped_ptr<PopupTimersController> popup_timers_controller_;
    scoped_ptr<base::OneShotTimer> quiet_mode_timer_;
    NotifierSettingsProvider* settings_provider_;
    std::vector<NotificationBlocker*> blockers_;

    // Queue for the notifications to delay the addition/updates when the message
    // center is visible.
    scoped_ptr<internal::ChangeQueue> notification_queue_;

    DISALLOW_COPY_AND_ASSIGN(MessageCenterImpl);
};

} // namespace message_center

#endif // UI_MESSAGE_CENTER_MESSAGE_CENTER_H_
